package com.shisan.tingshu.account.receiver;

import com.alibaba.fastjson.JSONObject;
import com.rabbitmq.client.Channel;
import com.shisan.tingshu.account.service.MqOpsService;
import com.shisan.tingshu.common.execption.ShisanException;
import com.shisan.tingshu.common.rabbit.constant.MqConst;
import com.shisan.tingshu.common.rabbit.service.RabbitService;
import com.shisan.tingshu.common.util.MD5;
import io.micrometer.common.util.StringUtils;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.annotation.Exchange;
import org.springframework.amqp.rabbit.annotation.Queue;
import org.springframework.amqp.rabbit.annotation.QueueBinding;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;

import java.io.IOException;
import java.util.Map;

@Component
@Slf4j
public class AccountInfoListener {

    @Autowired
    private MqOpsService mqOpsService;

    @Autowired
    private StringRedisTemplate redisTemplate;

    @Autowired
    private RabbitService rabbitService;

    // 通过注解就可以配置监听的队列、交换机、路由键，甚至可以配置持久化等参数
    @RabbitListener(bindings = @QueueBinding(
            value = @Queue(value = MqConst.QUEUE_USER_REGISTER, durable = "true"),
            exchange = @Exchange(value = MqConst.EXCHANGE_USER, durable = "true"),
            key = MqConst.ROUTING_USER_REGISTER
    ))
    @SneakyThrows // 可以绕开编译时的异常检查，但是真正运行期间如果出了异常依然会被抛出
    // 注意：
    // 引入的是 import org.springframework.amqp.core.Message;
    // 引入的是 import com.rabbitmq.client.Channel;
    // content是上游自定义发过来的消息，message是真正的消息，channel是通道
    public void listenUserAccountRegister(String content, Message message, Channel channel) {
        // 判断消息是否存在
        if (StringUtils.isEmpty(content)) {
            // 消息不存在，直接返回
            return;
        }
        // 消息存在，进行业务处理
        String msgMd5 = MD5.encrypt(content); // 使用MD5加密content，作为redis的key

        // 消费消息
        // 如果消费消息失败，消息还是会存在于队列中，下面的代码引入重试机制，重试次数用尽后，消息不再重新入队
        long deliveryTag = message.getMessageProperties().getDeliveryTag(); // 消息的唯一标识
        try {
            mqOpsService.userAccountRegister(content); // 会抛出ShisanException，被第一个异常处理器捕获
            // 手动应答消息（若不应答，消息会一直存在队列中）
            channel.basicAck(deliveryTag, false); // 抛出异常，被第二个异常处理器捕获
        } catch (ShisanException e) {
            // 不能只是简单地创建一个全局变量count来计数
            // 如果该service-account微服务被部署了多个实例，那么每个实例的count都会不一样，但实际上应该多个实例共用一个count
            // 所以需要使用redis来计数
            // 第一次重试时，msgRetryKey 不存在，创建后count的值是0
            // 第二次重试时，msgRetryKey 存在，count的值会自动+1
            String msgRetryKey = "msg:retry:" + msgMd5;
            Long count = redisTemplate.opsForValue().increment(msgRetryKey);
            // 三次重试
            if (count >= 3) { // 不能重试
                log.error("消息重试{}次失败，异常：{}", count, e.getMessage());
                channel.basicNack(deliveryTag, false, false); // 最后一个参数为true，表示不重新入队
                redisTemplate.delete(msgRetryKey); // 删除计数器
            } else {
                log.error("消息第{}次重试", count);
                channel.basicNack(deliveryTag, false, true); // 最后一个参数为false，表示重新入队
            }
        } catch (Exception e) {
            log.error("网络故障导致手动应答消息失败，异常：{}", e.getMessage());
            channel.basicNack(deliveryTag, false, false);
        }
    }

    @RabbitListener(bindings = @QueueBinding(
            value = @Queue(value = MqConst.QUEUE_ACCOUNT_UNLOCK, durable = "true"),
            exchange = @Exchange(value = MqConst.EXCHANGE_ACCOUNT, durable = "true"),
            key = MqConst.ROUTING_ACCOUNT_UNLOCK
    ))
    @SneakyThrows
    public void listenUserAccountUnlock(String orderNo, Message message, Channel channel) {
        // 判断消息是否存在
        if (StringUtils.isEmpty(orderNo)) {
            // 消息不存在，直接返回
            return;
        }
        // 消费消息
        long deliveryTag = message.getMessageProperties().getDeliveryTag(); // 消息的唯一标识
        try {
            mqOpsService.listenUserAccountUnlock(orderNo);
            // 向本地消息表发送签收消息
            rabbitService.sendMessage(MqConst.EXCHANGE_LOCAL_MSG, MqConst.ROUTING_LOCAL_MSG, orderNo);
            channel.basicAck(deliveryTag, false);
        } catch (ShisanException e) {
            String msgRetryKey = "msg:retry:" + orderNo;
            Long count = redisTemplate.opsForValue().increment(msgRetryKey);
            // 三次重试
            if (count >= 3) { // 不能重试
                log.error("消息重试{}次失败，异常：{}", count, e.getMessage());
                channel.basicNack(deliveryTag, false, false);
                redisTemplate.delete(msgRetryKey); // 删除计数器
            } else {
                log.error("消息第{}次重试", count);
                channel.basicNack(deliveryTag, false, true);
            }
        } catch (Exception e) {
            log.error("网络故障导致手动应答消息失败，异常：{}", e.getMessage());
            channel.basicNack(deliveryTag, false, false);
        }
    }

    @RabbitListener(bindings = @QueueBinding(
            value = @Queue(value = MqConst.QUEUE_ACCOUNT_MINUS, durable = "true"),
            exchange = @Exchange(value = MqConst.EXCHANGE_ACCOUNT, durable = "true"),
            key = MqConst.ROUTING_ACCOUNT_MINUS
    ))
    @SneakyThrows
    public void listenUserAccountMinus(String orderNo, Message message, Channel channel) {
        // 判断消息是否存在
        if (StringUtils.isEmpty(orderNo)) {
            // 消息不存在，直接返回
            return;
        }

        // 消费消息
        long deliveryTag = message.getMessageProperties().getDeliveryTag();
        try {
            mqOpsService.listenUserAccountMinus(orderNo);
            // 向本地消息表发送签收消息
            rabbitService.sendMessage(MqConst.EXCHANGE_LOCAL_MSG, MqConst.ROUTING_LOCAL_MSG, orderNo);
            channel.basicAck(deliveryTag, false);
        } catch (ShisanException e) {
            String msgRetryKey = "msg:retry:" + orderNo;
            Long count = redisTemplate.opsForValue().increment(msgRetryKey);
            // 三次重试
            if (count >= 3) { // 不能重试
                log.error("消息重试{}次失败，异常：{}", count, e.getMessage());
                channel.basicNack(deliveryTag, false, false);
                redisTemplate.delete(msgRetryKey); // 删除计数器
            } else {
                log.error("消息第{}次重试", count);
                channel.basicNack(deliveryTag, false, true);
            }
        } catch (Exception e) {
            log.error("网络故障导致手动应答消息失败，异常：{}", e.getMessage());
            channel.basicNack(deliveryTag, false, false);
        }
    }

    @RabbitListener(bindings = @QueueBinding(value = @Queue(value = MqConst.QUEUE_RECHARGE_PAY_SUCCESS, durable = "true"),
            exchange = @Exchange(value = MqConst.EXCHANGE_ORDER, durable = "true"),
            key = MqConst.ROUTING_RECHARGE_PAY_SUCCESS))
    @SneakyThrows
    public void listenUserAccountOpsRecharge(String content, Message message, Channel channel) {

        // 1.判断消息是否存在
        if (org.springframework.util.StringUtils.isEmpty(content)) {
            return;  // 不用消费
        }
        // 2.处理消息
        Map<String, String> map = JSONObject.parseObject(content, Map.class);
        String orderNo = map.get("orderNo");
        String userId = map.get("userId");
        String msgMd5 = MD5.encrypt(orderNo);
        // 3.消费消息
        long deliveryTag = message.getMessageProperties().getDeliveryTag();
        try {
            mqOpsService.userAccountOpsRecharge(userId, orderNo);
            // 4.手动应答消息（将消息从队列中删除掉）
            channel.basicAck(deliveryTag, false);
        } catch (ShisanException e) {
            String msgRetryKey = "msg:retry:" + msgMd5;
            Long count = redisTemplate.opsForValue().increment(msgRetryKey);
            // 三次重试
            if (count >= 3) {
                log.error("消息已经到达了重试{}次，请人工排查错误原因：{}", count, e.getMessage());
                // 不能重试
                channel.basicNack(deliveryTag, false, false);
                redisTemplate.delete(msgRetryKey);
            } else {
                log.info("消息重试{}次", count);
                channel.basicNack(deliveryTag, false, true);
            }
        } catch (Exception e) {
            log.error("签收消息时网络出现了故障，异常原因：{}", e.getMessage());
            channel.basicNack(deliveryTag, false, false);
        }
    }
}